#pragma once

#include "comm.hpp"

#include "boost/algorithm/string.hpp"

namespace ns_util
{
    const std::string temp_path = "./temp/";

    class TimeUtil
    {
    public:
        // 函数gettimeofday
        // 第一个参数是获取到的时间戳 ———— 从1970.1.1开始的秒数
        // 第二个参数是时区(不关心nullptr)
        static std::string GetTimeStamp()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec);
        }

        static std::string GetTimeMs()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 1000);
        }
    };

    // 路径工具
    class PathUtil
    {
    public:
        // 添加文件后缀
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }
        /*---编译时需要的文件---*/
        // 对文件添加源后缀
        static std::string SRC(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }

        // 对文件添加执行后缀
        static std::string EXE(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }

        // 对文件添加错误后缀
        static std::string Compiler_error(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compile_error");
        }

        /*---运行时需要的文件---*/
        // 对文件添加标准输入后缀
        static std::string IN(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }

        // 对文件添加标准输出后缀
        static std::string OUT(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }

        // 对文件添加标准错误后缀
        static std::string ERROR(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };

    // 文件工具
    class FileUtil
    {
    public:
        // 判断文件是否存在
        // 调用stat函数 ———— 通过文件路径可以获取文件的状态
        static bool IsFileExict(const std::string &path_name)
        {
            struct stat st;
            if (stat(path_name.c_str(), &st) == -1)
            {
                return false;
            }
            return true;
        }

        static std::string UniqFileName()
        {
            // 根据毫秒级时间戳 + 原子性自增来确保文件名统一
            std::atomic_uint _id(0);
            std::string ms = TimeUtil::GetTimeMs();
            std::string uniq_id = std::to_string(++_id);
            return ms + "_" + uniq_id;
        }

        // target 文件路径 content 写入的内容
        static bool WriteToFile(const std::string &target, const std::string &content)
        {
            std::ofstream out(target);
            if (!out.is_open())
            {
                return false;
            }
            out.write(content.c_str(), content.size());
            out.close();
            return true;
        }

        // target 文件路径 content 读取的内容 keep 是否保留换行符
        static bool ReadFromFile(const std::string &target, std::string *content, bool keep = false)
        {
            std::ifstream in(target);
            if (!in.is_open())
            {
                return false;
            }
            std::string line;
            // getline按行读取时不会读取分行符
            // getline内部重载了强制类型转换 ———— 当读取结束时返回false
            while (getline(in, line))
            {
                (*content) += line;
                (*content) += (keep ? "\n" : "");
            }
            in.close();
            return true;
        }
    };

    // 字符串工具
    class StringUtil
    {
    public:
        static void SplitString(const std::string &str, std::vector<std::string> *out, std::string sep)
        {
            boost::split(*out, str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };
}